import java.util.Comparator;
import java.util.Map;


public class ComparaModelos implements Comparator {
	private Map<String, Diagrama> mapaDiagrama;

	public ComparaModelos(Map<String, Diagrama> mapaDiagrama) {
		this.mapaDiagrama = mapaDiagrama;
	}

	public int compare(Object o1, Object o2) {
		int retorno;
		
		String nome1 = (String) o1;
		String nome2 = (String) o2;
		
		Diagrama d1 = this.mapaDiagrama.get(nome1);
		Diagrama d2 = this.mapaDiagrama.get(nome2);
		
		double pto1 = d1.getTotalPtos();
		double pto2 = d2.getTotalPtos();
		
		if (pto1 > pto2)
			retorno = -1;
		else if (pto1 == pto2)
			retorno = 0;
		else
			retorno = 1;
		
		return retorno;
	}
}
